package experiment;

import java.util.ArrayList;

import model.trajectory.TargetModel;
import model.trajectory.TrajectoryModel;

public class Dataset 
{
	private String name;
	private int difficulty;
	
	public ArrayList<TargetModel> trajectories;
	public ArrayList<Question> questions;
	
	public Dataset(String name, String filePath, int difficulty)
	{
		this.name = name;
		this.difficulty = difficulty;
		this.trajectories = new ArrayList<TargetModel>();
		this.questions = new ArrayList<Question>();
	}

	public void addTrajectory(TargetModel trajectory)
	{
		this.trajectories.add(trajectory);
	}
	
	public void addQuestion(Question question)
	{
		this.questions.add(question);
	}
	public String getName() {
		return name;
	}

	public int getDifficulty() {
		return difficulty;
	}

	public void setDifficulty(int difficulty) {
		this.difficulty = difficulty;
	}
	
	public String toString()
	{
		return "("+this.name + ";" + this.difficulty + ")";
	}
}
